#!/bin/sh
PYTHON=${PYTHON:-"python"}

# 单机单卡测试
# python main_eff.py -e -a 'mobilenet_v2' --pretrained data

# 单机多卡训练
$PYTHON main_eff.py data -a 'efficientnet_b0' --dist-url 'tcp://127.0.0.1:13201'\
--multiprocessing-distributed --world-size 1 --rank 3 \
--resume 'checkpoints/checkpoint89.pth.tar' --log-file 'log89' --epochs 120

# 单机单卡训练
# $PYTHON main_eff.py -b 128 -a 'efficientnet_b0' data --resume 'checkpoints/checkpoint13.pth.tar' --gpu 3